package com.lxy.htmlparse.parse;

import com.lxy.htmlparse.dao.PicInfoDao;
import com.lxy.htmlparse.empty.PicMinInfo;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import org.springframework.stereotype.Service;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.List;

/**
 * @Author: lxy
 * @Date: 2019/6/20 15:18
 * @Desc: describe this class
 */
@Component
public class DownloadPic {
    private static Logger logger = LoggerFactory.getLogger(WeiboHmtlParse.class);

    @Autowired
    private PicInfoDao picInfoDao;
    private String prefixUrl = "https://wx1.sinaimg.cn/bmiddle/";

    @Value("${img.store.path}")
    private String filePath;

    public void downloadPic() {
        List<PicMinInfo> pics = picInfoDao.findAll();
        for (PicMinInfo pic : pics) {
            String minUrl = pic.getMinUrl();
            String[] split = minUrl.split("\\/");
            String url = split[split.length - 1];
            String reqUrl = prefixUrl + url;
            try {
                getPicFile(reqUrl, url);
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        logger.info("下载完毕，一共下载{}张图片", pics.size());

    }

    private void getPicFile(String urls, String picName) throws Exception {
        //new一个URL对象
        URL url = new URL(urls);
        //打开链接
        HttpURLConnection conn = (HttpURLConnection) url.openConnection();
        //设置请求方式为"GET"
        conn.setRequestMethod("GET");
        //超时响应时间为5秒
        conn.setConnectTimeout(5 * 1000);
        //通过输入流获取图片数据
        InputStream inStream = conn.getInputStream();
        //得到图片的二进制数据，以二进制封装得到数据，具有通用性
        byte[] data = readInputStream(inStream);
        //new一个文件对象用来保存图片，默认保存当前工程根目录
        File imageFile = new File(filePath + picName);
        if (!imageFile.getParentFile().exists()) {
            imageFile.getParentFile().mkdirs();
        }
        //创建输出流
        FileOutputStream outStream = new FileOutputStream(imageFile);
        //写入数据
        outStream.write(data);
        //关闭输出流
        outStream.close();
    }

    public static byte[] readInputStream(InputStream inStream) throws Exception {
        ByteArrayOutputStream outStream = new ByteArrayOutputStream();
        //创建一个Buffer字符串
        byte[] buffer = new byte[1024];
        //每次读取的字符串长度，如果为-1，代表全部读取完毕
        int len = 0;
        //使用一个输入流从buffer里把数据读取出来
        while ((len = inStream.read(buffer)) != -1) {
            //用输出流往buffer里写入数据，中间参数代表从哪个位置开始读，len代表读取的长度
            outStream.write(buffer, 0, len);
        }
        //关闭输入流
        inStream.close();
        //把outStream里的数据写入内存
        return outStream.toByteArray();
    }
}

